package com.tuniu.agents.tools.flights.enums;

import com.tuniu.agents.tools.flights.domestic.oneway.FlightLowPriceQueryTool;
import com.tuniu.agents.tools.flights.domestic.oneway.FlightTimeQueryTool;
import com.tuniu.agents.tools.flights.domestic.oneway.FlightTransferQueryTool;

public enum FlightToolEnum {
    FLIGHT_TIME_TOOL_LIST(FlightTimeQueryTool.TOOLNAME,
            "flight-selection",
            "flight.time.tool.before.start.search,flight.time.tool.before.departure.time,flight.time.tool.before.arrival.time,flight.time.tool.before.de,flight.time.tool.before.effect.flight",
            "flight.time.tool.inner.departure.no.airport,flight.time.tool.inner.arrival.no.airport",
            ""),
    FLIGHT_UTILS_LIST(FlightTimeQueryTool.TOOLNAME,
            "",
            "flight.util.list.before.no.date,flight.util.list.before.have.date",
            "",
            "flight.util.list.after.no.num,flight.util.list.after.have.num"),
    FLIGHT_LOW_PRICE_LIST(FlightLowPriceQueryTool.TOOLNAME,
            "flight-selection",
            "flight.low.price.tool.before.start.search",
            "flight.time.tool.inner.departure.no.airport,flight.time.tool.inner.arrival.no.airport",
            ""),

    FLIGHT_TRANSFER_LIST(FlightTransferQueryTool.TOOLNAME,
            "flight-selection",
            "flight.transfer.tool.before.start.search",
            "flight.time.tool.inner.departure.no.airport,flight.time.tool.inner",
            ""),

    ROUND_FLIGHT_LOW_PRICE_LIST(FlightLowPriceQueryTool.TOOLNAME,
            "flight-selection",
            "around.flight.low.price.tool.before.start.search",
            "",
            ""),
    ;

    private String toolName;

    private String promptTemplateId;

    private String before;

    private String inner;

    private String after;

    FlightToolEnum(String toolName, String promptTemplateId, String before, String inner, String after) {
        this.toolName = toolName;
        this.promptTemplateId = promptTemplateId;
        this.before = before;
        this.inner = inner;
        this.after = after;
    }


    public String getToolName() {
        return toolName;
    }

    public String getPromptTemplateId() {
        return promptTemplateId;
    }

    public String getBefore() {
        return before;
    }

    public String getAfter() {
        return after;
    }

    public String getInner() {
        return inner;
    }
}